import numpy as np

# 创建示例数组
arr = np.array(
    [
        [1, 2, 3],
        [4, 5, 6],
        [7, 8, 9],
    ]
)

# where查找: 返回[行],[列]
idx = np.where(arr == 5)  # --OK
rows, cols = np.where(arr == 5)  # -OK
row_value = idx[0].item()
col_value = idx[1].item()
print(f"行{idx[0]}, 列{idx[1]}, row值{row_value},col列值{col_value}")
"""行[1], 列[1], row值1,col列值1"""

# 确认是否相等
idx_value = arr[idx]
if idx_value != 5:
    print("未找到")
else:
    print("找到")

# 3.查找不存在的值
idx = np.where(arr == 10)
print(idx)
""" 
(array([], dtype=int64), array([], dtype=int64))

len(arr[idx]) == 0 表示没有找到
"""